﻿//========================================
// All rights reserved (C) 2017-2100
// CLR 版本：	4.0.30319.42000
// 命名空间:	SFJDService
// 文 件 名：	FileReceive2
// 作    者：	WooG
// 联系方式：	wooglle@163.com
// 创建时间：	2018/7/10 23:07:57
//========================================
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Text;

namespace SFJDService
{
    class FileReceive2
    {
        internal static List<ClassTransCSInfo> tasks = new List<ClassTransCSInfo>(3);
        static int maxLength = 1400; //文件发送时的缓冲区大小
        Socket socketClient;
        private static Thread threadWatch = null;
        private static Socket socketWatch = null;

        public FileReceive2()
        {
            if (Runtime.onReciving)
            {
                return;
            }
            else
            {
                BeginListening();
            }
        }

        /// <summary>
        /// 开始监听
        /// </summary>
        public void BeginListening()
        {
            Runtime.onReciving = true;
            try
            {
                socketWatch = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);//创建服务端负责监听的套接字，参数（使用IPV4协议，使用流式连接，使用Tcp协议传输数据）
                IPEndPoint endpoint = new IPEndPoint(this.myIp, Runtime.portFile);//创建包含Ip和port的网络节点对象
                socketWatch.Bind(endpoint);//将负责监听的套接字绑定到唯一的Ip和端口上
                socketWatch.Listen(30);//设置监听队列的长度
                threadWatch = new Thread(ListeningConnect);//创建负责监听的线程，并传入监听方法
                threadWatch.IsBackground = true;//设置为后台线程
                threadWatch.Start();//开始线程
                Runtime.Printf("开始监听：" + endpoint.Address + ":" + endpoint.Port);
            }
            catch (Exception e)
            {
                throw new Exception(e.ToString());
            }
            finally
            {
                if (socketWatch != null && socketWatch.Connected)
                {
                    socketWatch.Close();
                    socketWatch.Dispose();
                }
                Runtime.onReciving = false;
            }
        }

        //// <summary>
        /// 接收远程主机发送的数据
        /// </summary>
        /// <param name="socket">要接收数据且已经连接到远程主机的 socket</param>
        /// <param name="buffer">接收数据的缓冲区</param>
        /// <param name="outTime">接收数据的超时时间，以秒为单位，可以精确到微秒</param>
        /// <returns>0:接收数据成功；-1:超时；-2:接收数据出现错误；-3:接收数据时出现异常</returns>
        /// <remarks >
        /// 1、当 outTime 指定为-1时，将一直等待直到有数据需要接收；
        /// 2、需要接收的数据的长度，由 buffer 的长度决定。
        /// </remarks>
        public static int RecvData(Socket socket, byte[] buffer, int outTime)
        {
            if (socket == null || socket.Connected == false)
            {
                throw new ArgumentException("参数socket 为null，或者未连接到远程计算机");
            }
            if (buffer == null || buffer.Length == 0)
            {
                throw new ArgumentException("参数buffer 为null ,或者长度为 0");
            }
            buffer.Initialize();
            int left = buffer.Length;
            int curRcv = 0;
            int flag = 0;

            try
            {
                while (true)
                {
                    if (socket.Poll(outTime * 1000000, SelectMode.SelectRead) == true)
                    {        // 已经有数据等待接收
                        curRcv = socket.Receive(buffer, curRcv, left, SocketFlags.None);
                        left -= curRcv;
                        if (left == 0)
                        {                                    // 数据已经全部接收 
                            flag = 0;
                            break;
                        }
                        else
                        {
                            if (curRcv > 0)
                            {                                // 数据已经部分接收
                                continue;
                            }
                            else
                            {                                            // 出现错误
                                flag = -2;
                                break;
                            }
                        }
                    }
                    else
                    {                                                    // 超时退出
                        flag = -1;
                        break;
                    }
                }
            }
            catch (SocketException e)
            {
                Runtime.Printf(e.ToString());
                flag = -3;
            }
            return flag;
        }

        //// <summary>
        /// 接收远程主机发送的数据
        /// </summary>
        /// <param name="socket">要接收数据且已经连接到远程主机的 socket</param>
        /// <param name="buffer">存储接收到的数据的字符串</param>
        /// <param name="bufferLen">待接收的数据的长度</param>
        /// <param name="outTime">接收数据的超时时间，以秒为单位，可以精确到微秒</param>
        /// <returns>0:接收数据成功；-1:超时；-2:接收数据出现错误；-3:接收数据时出现异常</returns>
        /// <remarks >
        /// 当 outTime 指定为-1时，将一直等待直到有数据需要接收；
        /// </remarks>
        public static int RecvData(Socket socket, string buffer, int bufferLen, int outTime)
        {
            if (bufferLen <= 0)
            {
                throw new ArgumentException("存储待接收数据的缓冲区长度必须大于0");
            }
            byte[] tmp = new byte[bufferLen];
            int flag = 0;
            if ((flag = RecvData(socket, tmp, outTime)) == 0)
            {
                buffer = System.Text.Encoding.Default.GetString(tmp);
            }
            return flag;
        }

        //// <summary>
        /// 接收远程主机发送的文件
        /// </summary>
        /// <param name="socket">待接收数据且已经连接到远程主机的 socket</param>
        /// <param name="fileName">保存接收到的数据的文件名</param>
        /// <param name="fileLength" >待接收的文件的长度</param>
        /// <param name="maxBufferLength">接收文件时最大的缓冲区大小</param>
        /// <param name="outTime">接受缓冲区数据的超时时间</param>
        /// <returns>0:接收文件成功；-1:超时；-2:接收文件出现错误；-3:接收文件出现异常；-4:写入接收文件发生错误</returns>
        /// <remarks >
        /// 当 outTime 指定为-1时，将一直等待直到有数据需要接收
        /// </remarks>
        public static int RecvFile(Socket socket, string fileName, long fileLength, int maxBufferLength, int outTime)
        {
            if (fileName == null || maxBufferLength <= 0)
            {
                throw new ArgumentException("保存接收数据的文件名称为空或发送缓冲区的大小设置不正确.");
            }

            int flag = 0;
            try
            {
                FileStream fs = new FileStream(fileName, FileMode.Create);
                byte[] buffer = null;

                if (fileLength <= maxBufferLength)
                {                /**//* 一次读取所传送的文件 */
                    buffer = new byte[fileLength];
                    if ((flag = RecvData(socket, buffer, outTime)) == 0)
                    {
                        fs.Write(buffer, 0, (int)fileLength);
                    }
                }
                else
                {                                        /**//* 循环读取网络数据，并写入文件 */
                    int rcvLen = maxBufferLength;
                    long leftLen = fileLength;                        //剩下未写入的数据
                    buffer = new byte[rcvLen];

                    while (leftLen != 0)
                    {
                        if ((flag = RecvData(socket, buffer, outTime)) < 0)
                        {
                            break;
                        }
                        fs.Write(buffer, 0, rcvLen);
                        leftLen -= rcvLen;
                        rcvLen = (maxBufferLength < leftLen) ? maxBufferLength : ((int)leftLen);
                    }
                }
                fs.Close();
            }
            catch (IOException e)
            {
                Runtime.Printf(e.ToString());
                flag = -4;
            }
            return flag;
        }
    }
}
